# 设置登陆拦截
from flask import g

"""
自定义一个装饰器，验证某些函数是否已经登陆，如果已经登陆了就继续访问
本质上就是一个拦截功能
"""


def login_required(func):
    def wrapper(*args, **kwargs):
        if g.user_id is not None:  # 已经登陆
            return func(*args, **kwargs)
        else:
            return {"message": "Invalid Token"}

    return wrapper
